package com.zzh.util.excel.registry;



import com.zzh.util.excel.enums.ExportExcelClassEnumInterface;

import java.util.HashMap;
import java.util.Map;

public class ExportExcelClassEnumRegistry {
    private static final Map<String, ExportExcelClassEnumInterface> CODE_TO_ENUM = new HashMap<>();

    // 注册枚举
    public static void register(ExportExcelClassEnumInterface exportEnum) {
        CODE_TO_ENUM.put(exportEnum.getCode(), exportEnum);
    }

    // 查找枚举
    public static ExportExcelClassEnumInterface getEnum(String code) {
        ExportExcelClassEnumInterface exportEnum = CODE_TO_ENUM.get(code);
        if (exportEnum == null) {
            throw new IllegalArgumentException("No matching enum for code: " + code);
        }
        return exportEnum;
    }
}
